FROM openanolis/anolisos:23

LABEL \
    maintainer="OpenAnolis AI SIG" \
    org.opencontainers.image.title="pytorch-intel" \
    org.opencontainers.image.version="2.0.1-23-ipex"

RUN dnf install anolis-epao-release -y
RUN dnf install pytorch -y

ARG TARGETARCH
RUN if [ "${TARGETARCH}" = "amd64" ]; \
then dnf install intel-extension-for-pytorch torch-vision -y ; \
fi

ENTRYPOINT ["/bin/bash"]
